def welcomer():
    print('-' * 20)
    print('欢迎登陆学员管理系统')
    print('1：添加学员')
    print('2：删除学员')
    print('3：修改学员信息')
    print('4：查询学员信息')
    print('5：显示所有学员信息')
    print('6：退出系统')
    print('-' * 20)


def add_info():
    """
    关于学员信息的增删改查功能操作是同一个列表
    1、接收用户输入学员信息
    2、判断：
       学员已存在，提示信息
       不存在，添加到字典（一个学员），然后字典添加到列表（多个学员）
    :return:
    """
    print('------添加学员--------')
    new_id = input('请输入学号:')
    new_name = input('请输入姓名:')
    new_tel = input('请输入电话:')
    # 若要使用全局变量(info列表)，用global修饰
    global info
    for e in info:
        # 列表中的每一个元素都是字典
        if new_id == e['id'] or new_name == e['name']:
            print('用户已存在')
            return
    # 不存在的情况，要添加到列表中
    # else与for配对，当for正常结束会执行else后的代码，不正常结束就不会执行
    dict_info = {'id': new_id, 'name': new_name, 'tel': new_tel}
    info.append(dict_info)
    print(f'添加成功！学员信息：{info}')


def del_info(del_name):
    # 存在，则删除
    for e in info:
        if del_name == e['name']:
            info.remove(e)
            print('删除成功')
            break
    # 不存在，提示
    # for若正常结束会执行else内的代码，不正常（break，return）结束，就不会执行
    else:
        print('要删除的学员不存在')
    # print(info)


def modify_info():
    for e in info:
        if modify_name == e['name']:
            e['tel'] = new_tel
            print(f'修改成功{info}')
            break
    else:
        print('用户不存在')



def select_all():
    pass


def select_info():
    pass


# 所有的功能都使用同一个列表（存放所有学员）
info = []
while True:
    # 欢迎界面
    welcomer()
    # 用户选择功能
    user_choose = input('请输入需要的功能序号：')
    # 根据用户选择，进入不同的功能
    if user_choose == '1':
        # 添加学员
        add_info()
    elif user_choose == '2':
        # 删除学员
        del_name = input('请输入要删除的学员姓名:')
        del_info(del_name)
    elif user_choose == '3':
        modify_name = input('请输入要修改的学员姓名:')
        new_tel = input('请输入要修改的信息:')
        modify_info()
    elif user_choose == '4':
        select_info()
    elif user_choose == '5':
        select_all()
    elif user_choose == '6':
        flag = input('确定要退出吗？Yes or no:')
        if flag == 'yes':
            break
    else:
        print('输入有误！请重新输入')
